import torch
import os

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

# 数据集（训练集和测试集）划分比例
test_size_rate = 0.2

# 模型参数
batch_size = 32
num_epochs = 50
learning_rate = 0.001
input_dim = 13
hidden_dim = 256
num_layers = 2
model_dir = './'  # 模型保存目录
os.makedirs(model_dir, exist_ok=True)
